Is there any way to make each matrix column have only one value in python